#!/bin/bash
#SBATCH --job-name=mutr
#SBATCH --nodes=1
#SBATCH -n 16
#SBATCH --gres=gpu:1
#SBATCH --output=logs/mutr/0.log
#SBATCH -p a40
#SBATCH --nodelist=a40c01

source /share/home/22351121/anaconda3/etc/profile.d/conda.sh
source activate Mevis
export NCCL_IB_DISABLE=1

NUM_GPUS=1
DIST_URL=10.82.8.1
NNODES=4
NODE_RANK=0

python3 -m torch.distributed.launch --nproc_per_node $NUM_GPUS --nnodes $NNODES --node_rank $NODE_RANK --master_port 13325 --master_addr $DIST_URL --use_env \
    train.py \
    --me mutr \
    --with_box_refine \
    --binary \
    --batch_size 1 \
    --epochs 2 \
    --lr_drop 1 \
    --num_frames 5 \
    --backbone video_swin_t_p4w7 \
    --backbone_pretrained ./weights/swin_tiny_patch244_window877_kinetics400_1k.pth \
    --output_dir ./results/mutr
